

/*
@note 只负责代码的编译
*/
#pragma once

#include <iostream>
#include <sys/wait.h>
#include <unistd.h>
#include "comm/log.hpp"
#include "comm/util.hpp"
using std::endl, std::cout, std::cin;

namespace ns_cpmpiler
{
    using namespace ns_util;
    using namespace ns_log;
    class Compiler
    {
    public:
        Compiler() {}
        ~Compiler() {}
        /*
            @return `编译成功返回 true，否则返回false`
            @param fileName 编译的文件名
            @note
            ```
            if fileName == 1234
            1234 --> /temp/1234.cpp
            1234 --> /temp/1234.exe
            1234 --> /temp/1234.stderr
            ```
        */
        static bool Compile(const std::string &fileName)
        {
            pid_t pid = fork();
            if (pid < 0)
            {
                LOG(ERROR) << "内部错误" << std::endl;
                return false;
            }
            else if (pid == 0)
            {
                int _stderr = open(PathUtil::CompilerError(fileName).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_stderr < 0)
                {
                    LOG(ERROR) << "没有成功形成.stderr文件" << std::endl;
                    exit(1);
                }
                // 重定向标准错误到 _stderr
                dup2(_stderr, 2);
                // 程序替换并不影响进程的程序描述符表
                // 子进程: 调用编译器，执行相关的代码
                ::execlp("g++", "g++", "-o", PathUtil::Exe(fileName).c_str(),\
                         PathUtil::Src(fileName).c_str(), "-std=c++11", "-D", "COMPILER_LINE", nullptr);
                LOG(ERROR) << "启动编译器g++失败， 可能是参数错误" << std::endl;
                exit(2);
            }
            else
            {
                ::waitpid(pid, nullptr, 0);
                // * 验证编译是否成功
                if (FileUtil::isFileExists(PathUtil::Exe(fileName)))
                {
                    LOG(INFO) << PathUtil::Src(fileName) << "编译成功" << std::endl;
                    return true;
                }
            }

            LOG(ERROR) << "编译失败，没有形成可执行程序" << std::endl;
            return false;
        }
    private:
    };
}